from django.db.models import Q
from drf_yasg2.utils import swagger_auto_schema
from rest_framework import viewsets
from drf_yasg2 import openapi
from rest_framework.decorators import action

from PAA_ERP.utils.base_views import MultipleOpenMixin, MultipleDestroyMixin
from PAA_ERP.utils.pagination import GlobalPagination
from goods_info.models import GoodsCategoryModel
from goods_info.serializer.category_serializer import CategorySerializers


class CategoryView(viewsets.ModelViewSet, MultipleOpenMixin, MultipleDestroyMixin):
    """
    create:
    商品类别（货品）信息--新增

    商品类别（货品）信息新增, status: 201(成功), return: 新增商品类别（货品）信息信息

    destroy:
    商品类别（货品）信息--删除

    商品类别（货品）信息删除, status: 204(成功), return: None

    multiple_delete:
    商品类别（货品）信息--批量删除,必传参数：ids=[1,2,3,4...]

    商品类别（货品）信息批量删除, status: 204(成功), return: None

    multiple_open:
    商品类别（货品）信息--批量启用或者禁用,必传(json)参数：ids=[1,2,3,4...](列表中可以只有一个)，is_open=1/0
    {
        "ids":[1,2],
        "is_open":"0"
    }
    is_open=1表示禁用，is_open=0表示启用，
    商品类别（货品）信息批量启用或者禁用, status: 204(成功), return: None

    update:
    商品类别（货品）信息--修改,

    商品类别（货品）信息修改, status: 200(成功), return: 修改后的商品类别（货品）信息信息

    partial_update:
    商品类别（货品）信息--局部修改,可以传参任意属性的值，服务器会修改指定的属性值 商品类别（货品）信息局部修改,

    status: 200(成功), return: 修改后的商品类别（货品）信息信息

    list: 查询所有顶级类别
    商品类别（货品）信息--获取分页列表，可选get请求参数:pid，如果没有传，则查询顶级类别，否则查询指定pid下的子类别

    商品类别（货品）信息列表信息, status: 200(成功), return: 商品类别（货品）信息信息列表

    retrieve:
    查询某一个商品类别（货品）信息

    查询指定ID的商品类别（货品）信息, status: 200(成功), return: 用户商品类别（货品）信息
    """
    queryset = GoodsCategoryModel.objects.all()
    serializer_class = CategorySerializers

    def get_queryset(self):
        if self.action == "list":  # 判断是否查询顶级列表还是某个子列表
            pid = self.request.query_params.get('pid', None)
            if pid and pid != '0':
                return GoodsCategoryModel.objects.filter(parent__id=pid).all()
            else:
                return GoodsCategoryModel.objects.filter(parent__isnull=True).all()
        else:
            return GoodsCategoryModel.objects.all()

    pid_param = openapi.Parameter(name='pid', in_=openapi.IN_QUERY, description='父类别的id，可以不传查询所有父类别',
                                  type=openapi.TYPE_INTEGER)

    @swagger_auto_schema(manual_parameters=[pid_param],
                         operation_description='商品类别的列表')
    def list(self, request, *args, **kwargs):
        return super(CategoryView, self).list(request=request, *args, **kwargs)
